Variance Reduction for Expectations with Diffusion Teachers
扩散教师模型期望的方差减少
🧭 扩散模型下游任务梯度方差优化
NVIDIA · University of Toronto · Princeton University
CARV通过分层蒙特卡洛估计摊销昂贵前向计算，以2-3倍有效计算倍乘降低下游梯度方差
📌 工作参考价值（4 点）
- 在推荐系统使用扩散模型做内容生成或特征提取时，可采用廉价噪声重采样摊销昂贵前向计算的模式：将多次去噪步的共享计算提前完成，仅在最终步或采样时重新采样噪声，大幅降低推理/训练成本。
- 当目标函数涉及对时间步的期望时，可借鉴重要性采样+分层逆CDF，根据梯度方差分布动态调整时间步采样概率，在电商多模态模型（如视频生成）中减少约25%的噪声估计方差。
- 构建计算感知的方差核算框架，显式衡量不同随机源的方差贡献与计算开销，辅助决策哪些组件值得投入更多采样预算——这不限于扩散模型，Agent多步推理中的蒙特卡洛估计同样可应用。
- 在单步蒸馏等场景，若梯度方差已非瓶颈，大幅降低方差未必提升下游指标（如FID），提示业务中应优先诊断瓶颈后再投入方差优化资源。